plt.figure(figsize = (50,5)); sns.heatmap(weights.T.loc[['ridge_cv'],:].abs(),annot=True, linewidths=.5,cbar=False,xticklabels=False); r2_results2 = pd.DataFrame({'ridge_cv':ridge_cv.best_score_},index=['CV R2']) r2_results2.loc['Train R2', :] =[r2_score(y_train,ridge_cv.predict(X_train))] r2_results2.loc['lambda','ridge_cv']=ridge_cv.alpha_ r2_results2